from django_redis import get_redis_connection
from rest_framework import serializers

from .models import OAuthQQUser
from users.models import User
from .utils import OAuthQQ


class QQAuthUserSerializer(serializers.Serializer):
    """
    QQ登陆创建用户序列化器
    """
    # 是外界的request.data传过来的注册时请求信息
    access_token = serializers.CharField(label="操作凭证")
    mobile = serializers.RegexField(label="手机号", regex=r"^1[3-9]\d{9}$")
    password = serializers.CharField(label="密码", max_length=20, min_length=8)
    sms_code = serializers.CharField(label="短信验证码")

    def validate(self, data):
        # 检验access_token
        access_token = data["access_token"]
        # 获取身份凭证
        openid = OAuthQQ.check_save_user_token(access_token)
        if not openid:
            raise serializers.ValidationError("无效的access_token")
        # 将openid放在校验字典中 后面会使用
        data["openid"] = openid

        mobile = data["mobile"]
        sms_code = data["sms_code"]
        redis_conn = get_redis_connection("verify_codes")
        real_smss_code = redis_conn.get("sms_%s" % mobile)
        if real_smss_code.decode() != sms_code:
            raise serializers.ValidationError("短信验证码错误")

        # 如果用户存在 检查用户民和密码
        try:
            user = User.objects.get(mobile=mobile)
        except User.DoesNotExist:
            pass
        else:
            password = data["password"]
            if not user.check_password(password):
                raise serializers.ValidationError("密码错误")

            # 将认证后的user放进到校验的字典中 后续会使用
            data["user"] = user
        return data

    def create(self, validated_data):
        # 获取校验用户
        user = validated_data.get("user")
        if not user:
            user = User.objects.create_user(
                username=validated_data["mobile"],
                password=validated_data["password"],
                mobile=validated_data["mobile"]
            )
        # 将用户绑定openid
        OAuthQQUser.objects.create(
            openid=validated_data["openid"],
            user=user
        )
        # 返回用户数据
        return user
